import os
from openai import OpenAI

#from dotenv import load_dotenv, find_dotenv
#_ = load_dotenv(find_dotenv())  # 读取本地 .env 文件，里面定义了 OPENAI_API_KEY
#os.environ["OPENAI_API_KEY"] ="sk-e5MQbboyDcZ7Xic93ciJguf1RYUnDKVBeB6iaAqf1wfChvLr"
#os.environ["OPENAI_API_BASE"]="https://api.fe8.cn/v1"
#os.environ["OPENAI_BASE_URL"]="https://api.fe8.cn/v1"


messages = [
    {
    "role":"system",
    "content":"""如果名字并不确定，请使用某某先生/女士来代替。这里的“某某”可以是姓氏，也可以是名字。但不需要直观显示。
    请使用以下格式返回：
    名字：xxxx
    不需要其它内容"""
},{
    "role":"user",
    "content":"请提取下面句子中的名字:把这本书交给那个赵什么来着。"
}]

model = "gpt-3.5-turbo"
#model = "gpt-4o"
client = OpenAI(api_key=os.environ["OPENAI_API_KEY_ZHIHU"] ,base_url=os.environ["OPENAI_API_BASE_ZHIHU"])
response = client.chat.completions.create(messages=messages, model="gpt-3.5-turbo")
print(response.choices[0].message.content)
